/** * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * * See the NOTICE file distributed with this work for additional * information regarding copyright ownership. */ package org.sintef.thingml.resource.thingml.ui; /** * A CodeCompletionHelper can be used to derive completion proposals for partial * documents. It runs the parser generated by EMFText in a special mode (i.e., the * rememberExpectedElements mode). Based on the elements that are expected by the * parser for different regions in the document, valid proposals are computed. */ public class ThingmlCodeCompletionHelper { private org.sintef.thingml.resource.thingml.mopp.ThingmlAttributeValueProvider attributeValueProvider = new org.sintef.thingml.resource.thingml.mopp.ThingmlAttributeValueProvider(); private org.sintef.thingml.resource.thingml.IThingmlMetaInformation metaInformation = new org.sintef.thingml.resource.thingml.mopp.ThingmlMetaInformation(); /** * Computes a set of proposals for the given document assuming the cursor is at * 'cursorOffset'. The proposals are derived using the meta information, i.e., the * generated language plug-in. * * @param originalResource * @param content the documents content * @param cursorOffset * * @return */ public org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal[] computeCompletionProposals(org.sintef.thingml.resource.thingml.IThingmlTextResource originalResource, String content, int cursorOffset) { org.eclipse.emf.ecore.resource.ResourceSet resourceSet = new org.eclipse.emf.ecore.resource.impl.ResourceSetImpl(); // the shadow resource needs the same URI because reference resolvers may use the // URI to resolve external references org.sintef.thingml.resource.thingml.IThingmlTextResource resource = (org.sintef.thingml.resource.thingml.IThingmlTextResource) resourceSet.createResource(originalResource.getURI()); java.io.ByteArrayInputStream inputStream = new java.io.ByteArrayInputStream(content.getBytes()); org.sintef.thingml.resource.thingml.IThingmlMetaInformation metaInformation = resource.getMetaInformation(); org.sintef.thingml.resource.thingml.IThingmlTextParser parser = metaInformation.createParser(inputStream, null); org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[] expectedElements = parseToExpectedElements(parser, resource, cursorOffset); if (expectedElements == null) { return new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal[0]; } if (expectedElements.length == 0) { return new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal[0]; } java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedAfterCursor = java.util.Arrays.asList(getElementsExpectedAt(expectedElements, cursorOffset)); java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedBeforeCursor = java.util.Arrays.asList(getElementsExpectedAt(expectedElements, cursorOffset - 1)); setPrefixes(expectedAfterCursor, content, cursorOffset); setPrefixes(expectedBeforeCursor, content, cursorOffset); // First, we derive all possible proposals from the set of elements that are // expected at the cursor position. java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> allProposals = new java.util.LinkedHashSet<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(); java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> rightProposals = deriveProposals(expectedAfterCursor, content, resource, cursorOffset); java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> leftProposals = deriveProposals(expectedBeforeCursor, content, resource, cursorOffset - 1); removeKeywordsEndingBeforeIndex(leftProposals, cursorOffset); // Second, the set of left proposals (i.e., the ones before the cursor) is checked // for emptiness. If the set is empty, the right proposals (i.e., the ones after // the cursor) are also considered. If the set is not empty, the right proposal // are discarded, because it does not make sense to propose them until the element // before the cursor was completed. allProposals.addAll(leftProposals); // Count the proposals before the cursor that match the prefix int leftMatchingProposals = 0; for (org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal leftProposal : leftProposals) { if (leftProposal.getMatchesPrefix()) { leftMatchingProposals++; } } if (leftMatchingProposals == 0) { allProposals.addAll(rightProposals); } // Third, the proposals are sorted according to their relevance. Proposals that // matched the prefix are preferred over ones that did not. Finally, proposals are // sorted alphabetically. final java.util.List<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> sortedProposals = new java.util.ArrayList<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(allProposals); java.util.Collections.sort(sortedProposals); org.eclipse.emf.ecore.EObject root = null; if (!resource.getContents().isEmpty()) { root = resource.getContents().get(0); } for (org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal proposal : sortedProposals) { proposal.setRoot(root); } return sortedProposals.toArray(new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal[sortedProposals.size()]); } public org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[] parseToExpectedElements(org.sintef.thingml.resource.thingml.IThingmlTextParser parser, org.sintef.thingml.resource.thingml.IThingmlTextResource resource, int cursorOffset) { final java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedElements = parser.parseToExpectedElements(null, resource, cursorOffset); if (expectedElements == null) { return new org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[0]; } removeDuplicateEntries(expectedElements); removeInvalidEntriesAtEnd(expectedElements); return expectedElements.toArray(new org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[expectedElements.size()]); } /** * Removes all expected elements that refer to the same terminal and that start at * the same position. */ protected void removeDuplicateEntries(java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedElements) { int size = expectedElements.size(); // We split the list of expected elements into buckets where each bucket contains // the elements that start at the same position. java.util.Map<Integer, java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal>> map = new java.util.LinkedHashMap<Integer, java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal>>(); for (int i = 0; i < size; i++) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal elementAtIndex = expectedElements.get(i); int start1 = elementAtIndex.getStartExcludingHiddenTokens(); java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> list = map.get(start1); if (list == null) { list = new java.util.ArrayList<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal>(); map.put(start1, list); } list.add(elementAtIndex); } // Then, we remove all duplicate elements from each bucket individually. for (int position : map.keySet()) { java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> list = map.get(position); removeDuplicateEntriesFromBucket(list); } // After removing all duplicates, we merge the buckets. expectedElements.clear(); for (int position : map.keySet()) { java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> list = map.get(position); expectedElements.addAll(list); } } /** * Removes all expected elements that refer to the same terminal. Attention: This * method assumes that the given list of expected terminals contains only elements * that start at the same position. */ protected void removeDuplicateEntriesFromBucket(java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedElements) { int size = expectedElements.size(); for (int i = 0; i < size - 1; i++) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal elementAtIndex = expectedElements.get(i); org.sintef.thingml.resource.thingml.IThingmlExpectedElement terminal = elementAtIndex.getTerminal(); for (int j = i + 1; j < size;) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal elementAtNext = expectedElements.get(j); if (terminal.equals(elementAtNext.getTerminal())) { expectedElements.remove(j); size--; } else { j++; } } } } protected void removeInvalidEntriesAtEnd(java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedElements) { for (int i = 0; i < expectedElements.size() - 1;) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal elementAtIndex = expectedElements.get(i); org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal elementAtNext = expectedElements.get(i + 1); // If the two expected elements have a different parent in the syntax definition, // we must not discard the second element, because is probably stems from a parent // rule. org.sintef.thingml.resource.thingml.grammar.ThingmlSyntaxElement symtaxElementOfThis = elementAtIndex.getTerminal().getSymtaxElement(); org.sintef.thingml.resource.thingml.grammar.ThingmlSyntaxElement symtaxElementOfNext = elementAtNext.getTerminal().getSymtaxElement(); boolean differentParent = symtaxElementOfNext.getParent() != symtaxElementOfThis.getParent(); boolean sameStartExcludingHiddenTokens = elementAtIndex.getStartExcludingHiddenTokens() == elementAtNext.getStartExcludingHiddenTokens(); boolean differentFollowSet = elementAtIndex.getFollowSetID() != elementAtNext.getFollowSetID(); if (sameStartExcludingHiddenTokens && differentFollowSet && !differentParent) { expectedElements.remove(i + 1); } else { i++; } } } /** * Removes all proposals for keywords that end before the given index. */ protected void removeKeywordsEndingBeforeIndex(java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> proposals, int index) { java.util.List<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> toRemove = new java.util.ArrayList<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(); for (org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal proposal : proposals) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal = proposal.getExpectedTerminal(); org.sintef.thingml.resource.thingml.IThingmlExpectedElement terminal = expectedTerminal.getTerminal(); if (terminal instanceof org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedCsString) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedCsString csString = (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedCsString) terminal; int startExcludingHiddenTokens = expectedTerminal.getStartExcludingHiddenTokens(); if (startExcludingHiddenTokens + csString.getValue().length() - 1 < index) { toRemove.add(proposal); } } } proposals.removeAll(toRemove); } protected String findPrefix(java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedElements, org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedAtCursor, String content, int cursorOffset) { if (cursorOffset < 0) { return ""; } int end = 0; for (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedElement : expectedElements) { if (expectedElement == expectedAtCursor) { final int start = expectedElement.getStartExcludingHiddenTokens(); if (start >= 0 && start < Integer.MAX_VALUE) { end = start; } break; } } end = Math.min(end, cursorOffset); final String prefix = content.substring(end, Math.min(content.length(), cursorOffset)); return prefix; } protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> deriveProposals(java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedElements, String content, org.sintef.thingml.resource.thingml.IThingmlTextResource resource, int cursorOffset) { java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> resultSet = new java.util.LinkedHashSet<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(); for (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedElement : expectedElements) { resultSet.addAll(deriveProposals(expectedElement, content, resource, cursorOffset)); } return resultSet; } protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> deriveProposals(final org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, String content, org.sintef.thingml.resource.thingml.IThingmlTextResource resource, int cursorOffset) { org.sintef.thingml.resource.thingml.IThingmlExpectedElement expectedElement = (org.sintef.thingml.resource.thingml.IThingmlExpectedElement) expectedTerminal.getTerminal(); if (expectedElement instanceof org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedCsString) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedCsString csString = (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedCsString) expectedElement; return handleKeyword(expectedTerminal, csString, expectedTerminal.getPrefix()); } else if (expectedElement instanceof org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedBooleanTerminal) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedBooleanTerminal expectedBooleanTerminal = (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedBooleanTerminal) expectedElement; return handleBooleanTerminal(expectedTerminal, expectedBooleanTerminal, expectedTerminal.getPrefix()); } else if (expectedElement instanceof org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedEnumerationTerminal) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedEnumerationTerminal expectedEnumerationTerminal = (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedEnumerationTerminal) expectedElement; return handleEnumerationTerminal(expectedTerminal, expectedEnumerationTerminal, expectedTerminal.getPrefix()); } else if (expectedElement instanceof org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedStructuralFeature) { final org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedStructuralFeature expectedFeature = (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedStructuralFeature) expectedElement; final org.eclipse.emf.ecore.EStructuralFeature feature = expectedFeature.getFeature(); final org.eclipse.emf.ecore.EClassifier featureType = feature.getEType(); final org.eclipse.emf.ecore.EObject container = findCorrectContainer(expectedTerminal); // Here it gets really crazy. We need to modify the model in a way that reflects // the state the model would be in, if the expected terminal were present. After // computing the corresponding completion proposals, the original state of the // model is restored. This procedure is required, because different models can be // required for different completion situations. This can be particularly observed // when the user has not yet typed a character that starts an element to be // completed. final java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> proposals = new java.util.ArrayList<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(); expectedTerminal.materialize(new Runnable() { public void run() { if (feature instanceof org.eclipse.emf.ecore.EReference) { org.eclipse.emf.ecore.EReference reference = (org.eclipse.emf.ecore.EReference) feature; if (featureType instanceof org.eclipse.emf.ecore.EClass) { if (reference.isContainment()) { // the FOLLOW set should contain only non-containment references assert false; } else { proposals.addAll(handleNCReference(expectedTerminal, container, reference, expectedTerminal.getPrefix(), expectedFeature.getTokenName())); } } } else if (feature instanceof org.eclipse.emf.ecore.EAttribute) { org.eclipse.emf.ecore.EAttribute attribute = (org.eclipse.emf.ecore.EAttribute) feature; if (featureType instanceof org.eclipse.emf.ecore.EEnum) { org.eclipse.emf.ecore.EEnum enumType = (org.eclipse.emf.ecore.EEnum) featureType; proposals.addAll(handleEnumAttribute(expectedTerminal, expectedFeature, enumType, expectedTerminal.getPrefix(), container)); } else { // handle EAttributes (derive default value depending on the type of the // attribute, figure out token resolver, and call deResolve()) proposals.addAll(handleAttribute(expectedTerminal, expectedFeature, container, attribute, expectedTerminal.getPrefix())); } } else { // there should be no other subclass of EStructuralFeature assert false; } } }); // Return the proposals that were computed in the closure call. return proposals; } else { // there should be no other class implementing IExpectedElement assert false; } return java.util.Collections.emptyList(); } protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> handleEnumAttribute(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedStructuralFeature expectedFeature, org.eclipse.emf.ecore.EEnum enumType, String prefix, org.eclipse.emf.ecore.EObject container) { java.util.Collection<org.eclipse.emf.ecore.EEnumLiteral> enumLiterals = enumType.getELiterals(); java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> result = new java.util.LinkedHashSet<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(); for (org.eclipse.emf.ecore.EEnumLiteral literal : enumLiterals) { String unResolvedLiteral = literal.getLiteral(); // use token resolver to get de-resolved value of the literal org.sintef.thingml.resource.thingml.IThingmlTokenResolverFactory tokenResolverFactory = metaInformation.getTokenResolverFactory(); org.sintef.thingml.resource.thingml.IThingmlTokenResolver tokenResolver = tokenResolverFactory.createTokenResolver(expectedFeature.getTokenName()); String resolvedLiteral = tokenResolver.deResolve(unResolvedLiteral, expectedFeature.getFeature(), container); boolean matchesPrefix = matches(resolvedLiteral, prefix); result.add(new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal(expectedTerminal, resolvedLiteral, prefix, matchesPrefix, expectedFeature.getFeature(), container)); } return result; } protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> handleNCReference(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, org.eclipse.emf.ecore.EObject container, org.eclipse.emf.ecore.EReference reference, String prefix, String tokenName) { // proposals for non-containment references are derived by calling the reference // resolver switch in fuzzy mode. org.sintef.thingml.resource.thingml.IThingmlReferenceResolverSwitch resolverSwitch = metaInformation.getReferenceResolverSwitch(); org.sintef.thingml.resource.thingml.IThingmlTokenResolverFactory tokenResolverFactory = metaInformation.getTokenResolverFactory(); org.sintef.thingml.resource.thingml.IThingmlReferenceResolveResult<org.eclipse.emf.ecore.EObject> result = new org.sintef.thingml.resource.thingml.mopp.ThingmlReferenceResolveResult<org.eclipse.emf.ecore.EObject>(true); resolverSwitch.resolveFuzzy(prefix, container, reference, 0, result); java.util.Collection<org.sintef.thingml.resource.thingml.IThingmlReferenceMapping<org.eclipse.emf.ecore.EObject>> mappings = result.getMappings(); if (mappings != null) { java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> resultSet = new java.util.LinkedHashSet<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(); for (org.sintef.thingml.resource.thingml.IThingmlReferenceMapping<org.eclipse.emf.ecore.EObject> mapping : mappings) { org.eclipse.swt.graphics.Image image = null; if (mapping instanceof org.sintef.thingml.resource.thingml.mopp.ThingmlElementMapping<?>) { org.sintef.thingml.resource.thingml.mopp.ThingmlElementMapping<?> elementMapping = (org.sintef.thingml.resource.thingml.mopp.ThingmlElementMapping<?>) mapping; Object target = elementMapping.getTargetElement(); // de-resolve reference to obtain correct identifier org.sintef.thingml.resource.thingml.IThingmlTokenResolver tokenResolver = tokenResolverFactory.createTokenResolver(tokenName); final String identifier = tokenResolver.deResolve(elementMapping.getIdentifier(), reference, container); if (target instanceof org.eclipse.emf.ecore.EObject) { image = getImage((org.eclipse.emf.ecore.EObject) target); } boolean matchesPrefix = matches(identifier, prefix); resultSet.add(new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal(expectedTerminal, identifier, prefix, matchesPrefix, reference, container, image)); } } return resultSet; } return java.util.Collections.emptyList(); } protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> handleAttribute(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedStructuralFeature expectedFeature, org.eclipse.emf.ecore.EObject container, org.eclipse.emf.ecore.EAttribute attribute, String prefix) { java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> resultSet = new java.util.LinkedHashSet<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(); Object[] defaultValues = attributeValueProvider.getDefaultValues(attribute); if (defaultValues != null) { for (Object defaultValue : defaultValues) { if (defaultValue != null) { org.sintef.thingml.resource.thingml.IThingmlTokenResolverFactory tokenResolverFactory = metaInformation.getTokenResolverFactory(); String tokenName = expectedFeature.getTokenName(); if (tokenName != null) { org.sintef.thingml.resource.thingml.IThingmlTokenResolver tokenResolver = tokenResolverFactory.createTokenResolver(tokenName); if (tokenResolver != null) { String defaultValueAsString = tokenResolver.deResolve(defaultValue, attribute, container); boolean matchesPrefix = matches(defaultValueAsString, prefix); resultSet.add(new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal(expectedTerminal, defaultValueAsString, prefix, matchesPrefix, expectedFeature.getFeature(), container)); } } } } } return resultSet; } /** * Creates a set of completion proposals from the given keyword. */ protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> handleKeyword(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedCsString csString, String prefix) { String proposal = csString.getValue(); boolean matchesPrefix = matches(proposal, prefix); return java.util.Collections.singleton(new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal(expectedTerminal, proposal, prefix, matchesPrefix, null, null)); } /** * Creates a set of (two) completion proposals from the given boolean terminal. */ protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> handleBooleanTerminal(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedBooleanTerminal expectedBooleanTerminal, String prefix) { java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> result = new java.util.LinkedHashSet<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(2); org.sintef.thingml.resource.thingml.grammar.ThingmlBooleanTerminal booleanTerminal = expectedBooleanTerminal.getBooleanTerminal(); result.addAll(handleLiteral(expectedTerminal, booleanTerminal.getAttribute(), prefix, booleanTerminal.getTrueLiteral())); result.addAll(handleLiteral(expectedTerminal, booleanTerminal.getAttribute(), prefix, booleanTerminal.getFalseLiteral())); return result; } /** * Creates a set of completion proposals from the given enumeration terminal. For * each enumeration literal one proposal is created. */ protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> handleEnumerationTerminal(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedEnumerationTerminal expectedEnumerationTerminal, String prefix) { java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> result = new java.util.LinkedHashSet<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal>(2); org.sintef.thingml.resource.thingml.grammar.ThingmlEnumerationTerminal enumerationTerminal = expectedEnumerationTerminal.getEnumerationTerminal(); java.util.Map<String, String> literalMapping = enumerationTerminal.getLiteralMapping(); for (String literalName : literalMapping.keySet()) { result.addAll(handleLiteral(expectedTerminal, enumerationTerminal.getAttribute(), prefix, literalMapping.get(literalName))); } return result; } protected java.util.Collection<org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal> handleLiteral(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal, org.eclipse.emf.ecore.EAttribute attribute, String prefix, String literal) { if ("".equals(literal)) { return java.util.Collections.emptySet(); } boolean matchesPrefix = matches(literal, prefix); return java.util.Collections.singleton(new org.sintef.thingml.resource.thingml.ui.ThingmlCompletionProposal(expectedTerminal, literal, prefix, matchesPrefix, null, null)); } /** * Calculates the prefix for each given expected element. The prefix depends on * the current document content, the cursor position, and the position where the * element is expected. */ protected void setPrefixes(java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedElements, String content, int cursorOffset) { if (cursorOffset < 0) { return; } for (org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedElement : expectedElements) { String prefix = findPrefix(expectedElements, expectedElement, content, cursorOffset); expectedElement.setPrefix(prefix); } } public org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[] getElementsExpectedAt(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[] allExpectedElements, int cursorOffset) { java.util.List<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal> expectedAtCursor = new java.util.ArrayList<org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal>(); for (int i = 0; i < allExpectedElements.length; i++) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedElement = allExpectedElements[i]; int startIncludingHidden = expectedElement.getStartIncludingHiddenTokens(); int end = getEnd(allExpectedElements, i); if (cursorOffset >= startIncludingHidden && cursorOffset <= end) { expectedAtCursor.add(expectedElement); } } return expectedAtCursor.toArray(new org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[expectedAtCursor.size()]); } /** * Calculates the end index of the expected element at allExpectedElements[index]. * To determine the end, the subsequent expected elements from the array of all * expected elements are used. An element is considered to end one character * before the next elements starts. */ protected int getEnd(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal[] allExpectedElements, int indexInList) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal elementAtIndex = allExpectedElements[indexInList]; int startIncludingHidden = elementAtIndex.getStartIncludingHiddenTokens(); int startExcludingHidden = elementAtIndex.getStartExcludingHiddenTokens(); for (int i = indexInList + 1; i < allExpectedElements.length; i++) { org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal elementAtI = allExpectedElements[i]; int startIncludingHiddenForI = elementAtI.getStartIncludingHiddenTokens(); int startExcludingHiddenForI = elementAtI.getStartExcludingHiddenTokens(); if (startIncludingHidden != startIncludingHiddenForI || startExcludingHidden != startExcludingHiddenForI) { return startIncludingHiddenForI - 1; } } return Integer.MAX_VALUE; } /** * Checks whether the given proposed string matches the prefix. The two strings * are compared ignoring the case. The prefix is also considered to match if is a * camel case representation of the proposal. */ protected boolean matches(String proposal, String prefix) { if (proposal == null || prefix == null) { return false; } return (proposal.toLowerCase().startsWith(prefix.toLowerCase()) || org.sintef.thingml.resource.thingml.util.ThingmlStringUtil.matchCamelCase(prefix, proposal) != null) && !proposal.equals(prefix); } protected org.eclipse.swt.graphics.Image getImage(org.eclipse.emf.ecore.EObject element) { if (!org.eclipse.core.runtime.Platform.isRunning()) { return null; } org.eclipse.emf.edit.provider.ComposedAdapterFactory adapterFactory = new org.eclipse.emf.edit.provider.ComposedAdapterFactory(org.eclipse.emf.edit.provider.ComposedAdapterFactory.Descriptor.Registry.INSTANCE); adapterFactory.addAdapterFactory(new org.eclipse.emf.edit.provider.resource.ResourceItemProviderAdapterFactory()); adapterFactory.addAdapterFactory(new org.eclipse.emf.ecore.provider.EcoreItemProviderAdapterFactory()); adapterFactory.addAdapterFactory(new org.eclipse.emf.edit.provider.ReflectiveItemProviderAdapterFactory()); org.eclipse.emf.edit.ui.provider.AdapterFactoryLabelProvider labelProvider = new org.eclipse.emf.edit.ui.provider.AdapterFactoryLabelProvider(adapterFactory); return labelProvider.getImage(element); } protected org.eclipse.emf.ecore.EObject findCorrectContainer(org.sintef.thingml.resource.thingml.mopp.ThingmlExpectedTerminal expectedTerminal) { org.eclipse.emf.ecore.EObject container = expectedTerminal.getContainer(); org.eclipse.emf.ecore.EClass ruleMetaclass = expectedTerminal.getTerminal().getRuleMetaclass(); if (ruleMetaclass.isInstance(container)) { // container is correct for expected terminal return container; } // the container is wrong org.eclipse.emf.ecore.EObject parent = null; org.eclipse.emf.ecore.EObject previousParent = null; org.eclipse.emf.ecore.EObject correctContainer = null; org.eclipse.emf.ecore.EObject hookableParent = null; org.sintef.thingml.resource.thingml.grammar.ThingmlContainmentTrace containmentTrace = expectedTerminal.getContainmentTrace(); org.eclipse.emf.ecore.EClass startClass = containmentTrace.getStartClass(); org.sintef.thingml.resource.thingml.mopp.ThingmlContainedFeature currentLink = null; org.sintef.thingml.resource.thingml.mopp.ThingmlContainedFeature previousLink = null; org.sintef.thingml.resource.thingml.mopp.ThingmlContainedFeature[] containedFeatures = containmentTrace.getPath(); for (int i = 0; i < containedFeatures.length; i++) { currentLink = containedFeatures[i]; if (i > 0) { previousLink = containedFeatures[i - 1]; } org.eclipse.emf.ecore.EClass containerClass = currentLink.getContainerClass(); hookableParent = findHookParent(container, startClass, currentLink, parent); if (hookableParent != null) { // we found the correct parent break; } else { previousParent = parent; parent = containerClass.getEPackage().getEFactoryInstance().create(containerClass); if (parent != null) { if (previousParent == null) { // replace container for expectedTerminal with correctContainer correctContainer = parent; } else { // This assignment is only performed to get rid of a warning about a potential // null pointer access. Variable 'previousLink' cannot be null here, because it is // initialized for all loop iterations where 'i' is greather than 0 and for the // case where 'i' equals zero, this path is never executed, because // 'previousParent' is null in this case. org.sintef.thingml.resource.thingml.mopp.ThingmlContainedFeature link = previousLink; org.sintef.thingml.resource.thingml.util.ThingmlEObjectUtil.setFeature(parent, link.getFeature(), previousParent, false); } } } } if (correctContainer == null) { correctContainer = container; } if (currentLink == null) { return correctContainer; } hookableParent = findHookParent(container, startClass, currentLink, parent); final org.eclipse.emf.ecore.EObject finalHookableParent = hookableParent; final org.eclipse.emf.ecore.EStructuralFeature finalFeature = currentLink.getFeature(); final org.eclipse.emf.ecore.EObject finalParent = parent; if (parent != null && hookableParent != null) { expectedTerminal.setAttachmentCode(new Runnable() { public void run() { org.sintef.thingml.resource.thingml.util.ThingmlEObjectUtil.setFeature(finalHookableParent, finalFeature, finalParent, false); } }); } return correctContainer; } /** * Walks up the containment hierarchy to find an EObject that is able to hold * (contain) the given object. */ protected org.eclipse.emf.ecore.EObject findHookParent(org.eclipse.emf.ecore.EObject container, org.eclipse.emf.ecore.EClass startClass, org.sintef.thingml.resource.thingml.mopp.ThingmlContainedFeature currentLink, org.eclipse.emf.ecore.EObject object) { org.eclipse.emf.ecore.EClass containerClass = currentLink.getContainerClass(); while (container != null) { if (containerClass.isInstance(object)) { if (startClass.equals(container.eClass())) { return container; } } container = container.eContainer(); } return null; } }